import cv2

# 初始化摄像头，0表示默认摄像头
cap = cv2.VideoCapture(0)

# 检查摄像头是否成功打开
if not cap.isOpened():
    print("无法打开摄像头")
    exit()

# 设置图像分辨率（可选）
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)

# 图像保存路径和文件名前缀
save_path = "/home/orangepi/image_save/"
file_prefix = "image_"
count = 0

try:
    while True:
        # 捕获一帧图像
        ret, frame = cap.read()
        
        # 如果捕获成功，ret为True
        if not ret:
            print("无法获取帧")
            break
        
        # 保存图像
        filename = f"{save_path}{file_prefix}{count}.jpg"
        cv2.imwrite(filename, frame)
        print(f"保存图像: {filename}")
        count += 1
        
        # 显示图像（可选）
        cv2.imshow('Camera', frame)
        
        # 按'q'键退出循环
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
finally:
    # 释放摄像头并关闭窗口
    cap.release()
    cv2.destroyAllWindows()
